package com.ruoyi.common.utils;

import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import sun.misc.BASE64Encoder;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;

/**
 * 合成二维码工具类
 *
 * @author yr
 */
public class QRCodeUtil {

    /**
     * 图片的宽度
     */
    private static final int IMAGE_WIDTH = 350;
    /**
     * 图片的高度(需按实际内容高度进行调整)
     */
    private static final int IMAGE_HEIGHT = 350;
    /**
     * 二维码的宽度
     */
    private static final int QR_CODE_WIDTH = 150;
    /**
     * 二维码的宽度
     */
    private static final int QR_CODE_HEIGHT = 150;

    private static final String FORMAT_NAME = "JPG";

    public static BufferedImage createQrCode(String notes, String content) {
        return createQrCode(null,null,notes,content,null,280,280);
    }

    /**
     * @param imgLogo logo图片
     * @param title   头部标题
     * @param notes   备注（描述）
     * @param content 二维码内容
     * @param footer  底部文字
     */
    public static BufferedImage createQrCode(String imgLogo, String title, String notes, String content, String footer,
                                             int codeHeight,int codeWidth) {
        // 头部文字区域高度
        int titleHeight = 50;

        // 创建主模板图片
        BufferedImage image = new BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);
        Graphics2D main = image.createGraphics();
        // 设置图片的背景色
        main.setColor(Color.white); //白色
        main.fillRect(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT);

        // 动态高度
        int height = 0;

        //***********************页面头部 文字****************
        if(StrUtil.isNotBlank(title)){
            Graphics2D titleRight = image.createGraphics();
            // 设置字体颜色 black黑 white白
            titleRight.setColor(Color.black);
            // 设置字体
            Font titleFont = new Font("宋体", Font.BOLD, 25);
            titleRight.setFont(titleFont);
            titleRight.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
            // 居中 x开始的位置：（图片宽度-字体大小*字的个数）/2
            int x = (IMAGE_WIDTH - (titleFont.getSize() * title.length())) / 2;
            titleRight.drawString(title, x, (titleHeight) / 2 + 10);
            height += titleHeight;
        }

        //**********************中间文字部分*********
        Graphics2D centerWord = image.createGraphics();
        // 设置字体颜色，先设置颜色，再填充内容
        centerWord.setColor(Color.black);
        // 设置字体
        Font wordFont = new Font("宋体", Font.PLAIN, 15);
        centerWord.setFont(wordFont);
        centerWord.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
        String[] info = notes.split("-");
        for (String s : info) {
            // x开始的位置：（图片宽度-字体大小*字的个数）/2
            int strWidth = centerWord.getFontMetrics().stringWidth(s);
            // 总长度减去文字长度的一半  （居中显示）
            int startX = (IMAGE_WIDTH - strWidth) / 2;
            height += 20;
            centerWord.drawString(s, startX, height);
        }

        //***************插入二维码图片***********************************************
        if (codeWidth == 0) codeWidth = QR_CODE_WIDTH;
        if (codeHeight == 0) codeHeight = QR_CODE_HEIGHT;
        Graphics codePic = image.getGraphics();
        BufferedImage codeImg;
        QrConfig config = new QrConfig();
        config.setWidth(codeWidth);
        config.setHeight(codeHeight);
        if (StrUtil.isNotBlank(imgLogo)) {
            config.setImg(imgLogo);
        }
        codeImg = QrCodeUtil.generate(content, config);
        // 绘制二维码
        codePic.drawImage(codeImg, (IMAGE_WIDTH - codeWidth) / 2, ++height, codeWidth, codeHeight, null);
        codePic.dispose();

        //**********************底部公司名字*********
        if (StrUtil.isNotBlank(footer)){
            Graphics2D typeLeft = image.createGraphics();
            // 设置字体颜色
            typeLeft.setColor(Color.black);
            // 设置字体
            Font footerFont = new Font("宋体", Font.PLAIN, 10);
            typeLeft.setFont(footerFont);
            typeLeft.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
            // x开始的位置：（图片宽度-字体大小*字的个数）/2
            int startX = (IMAGE_WIDTH - (footerFont.getSize() * footer.length())) / 2;
            height += codeHeight;
            typeLeft.drawString(footer, startX, height);
        }

        //***************插入标志图片***********************************************
//        Graphics signPic = image.getGraphics();
//        BufferedImage signImg = null;
//        try {
//            signImg = ImageIO.read(new java.io.File(imgSign));
//        } catch (Exception e) {
//        }
//
//        if (signImg != null) {
//            signPic.drawImage(signImg, 0, 130, QR_CODE_WIDTH, QR_CODE_HEIGHT, null);
//            signPic.dispose();
//        }
        // 返回image方便后续处理是生成图片还是生成base64字符串
        return image;
    }

    // 生成图片文件
    public static void createImage(BufferedImage image, String fileLocation) {
        if (image != null) {
            try {
                ImageIO.write(image, "png", new File(fileLocation));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    // 获取图片base64数据
    public static String base64ImageString(BufferedImage image) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();//io流
        ImageIO.write(image, FORMAT_NAME, bos);//写入流中
        byte[] bytes = bos.toByteArray();//转换成字节
        BASE64Encoder encoder = new BASE64Encoder();
        String jpgBase64 = encoder.encodeBuffer(bytes).trim();//转换成base64串
        jpgBase64 = jpgBase64.replaceAll("\n", "").replaceAll("\r", "");//删除 \r\n
        return "data:image/jpg;base64," + jpgBase64;
    }

//    public static void main(String[] args) {
//        String notes = "测试-ABDabc151";
//        String content = "http://www.baidu.com/";
//        BufferedImage bufferedImage = createQrCode(notes, content);
//        createImage(bufferedImage, "D:/test.png");
//    }

}
